当我们想要给表格中的内容进行分类，一般人想到的是excel中的做法。那还有没有什么特别好的解决办法呢？小编觉得python中的文本聚类文本聚类不错，分类的速度比较快，而且不会以出现分类混乱的问题。小编整理了有关文本聚类用来分类的代码，分享给小伙伴们一起尝试一下。

主要有一下几个步骤：

切词

去除停用词

构建词袋空间VSM(vector space model)

TF-IDF构建词权重，这部我没有做，因为我的数据基本都是一类的，只是想细分，所以感觉不太适合，而且这个也有点难（捂脸）

使用K-means算法

下面开始代码部分：

 #引入基础库，在网上抄的代码，除了1、2、6，其他的可能用不到  import numpy as np  import pandas as pd  import re  import os  import codecs  import jieba  #打开文件，文件在桌面上，可以自行修改路径  f1=open("C:/Users/KangB/Desktop/wechat7/title.txt","r",encoding='GB2312',errors='ignore')  f2=open("C:/Users/KangB/Desktop/wechat7/title_fenci.txt",'w',encoding='GB2312',errors='ignore')  for line in f1:  seg_list = jieba.cut(line, cut_all=False)  f2.write((" ".join(seg_list)).replace("\t\t\t","\t"))  #print(w)  f1.close()  f2.close()  #取需要分词的内容  titles=open("C:/Users/KangB/Desktop/wechat7/title_fenci.txt",encoding='GB2312',errors='ignore').read().split('\n')  #查看内容，这里是一个list,list里面每个原素是分好的标题，查看下长度看有没有错误  #titles  #len(titles)  #构建停词函数，停词表是自己在网上搜的  def get_custom_stopwords(stop_words_file):  with open(stop_words_file,encoding='utf-8')as f:  stopwords=f.read()  stopwords_list=stopwords.split('\n')  custom_stopwords_list=[i for i in stopwords_list]  return custom_stopwords_list  #停用词函数调用  stop_words_file="C:/Users/KangB/Desktop/wechat7/stopwords.txt"  stopwords=get_custom_stopwords(stop_words_file)  #查看停用词，也是list格式  #stopwords  #构建词向量，也就是把分好的次去除停词转化成kmeans可以接受的形式  from sklearn.feature_extraction.text import CountVectorizer  count_vec=CountVectorizer(stop_words=stopwords)  km_matrix= count_vec.fit_transform(titles)  print(km_matrix.shape)  #查看词向量  #print(km_matrix.toarray())  #开始聚类啦  from sklearn.cluster import KMeans  num_clusters = 4 #聚为四类，可根据需要修改  km = KMeans(n_clusters=num_clusters)  km.fit(km_matrix)  clusters = km.labels_.tolist()  #查看聚类的结果，是list,这里省略，看看长度是不是和title一样就行啦  #len(clusters)  #最后把聚类结果写在一个新的txt里面  f3 =open("C:/Users/KangB/Desktop/wechat7/title_clusters.txt", 'w',encoding='GB2312',errors='ignore')  for i in clusters:  f3.write(str(i))  f3.write("\n")  f3.close()

最后把原始数据title和聚类结果title_clusters在一个excel里面不同列打开就可以啦。

不知道有没有小伙伴们试了文本聚类的分类方法，是不是跟小编说的一样好用呢？小编相信试过的小伙伴，肯定下次还会使用这种分类方法的。